python3 main.py data -e -a 'efficientnet-b3' --pretrained --gpu 0 --batch-size 128
